/*
 * Copyright 2020 eBlocker Open Source UG (haftungsbeschraenkt)
 *
 * Licensed under the EUPL, Version 1.2 or - as soon they will be
 * approved by the European Commission - subsequent versions of the EUPL
 * (the "License"); You may not use this work except in compliance with
 * the License. You may obtain a copy of the License at:
 *
 *   https://joinup.ec.europa.eu/page/eupl-text-11-12
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */
package org.eblocker.server.icap.transaction.processor;

import org.eblocker.server.common.RequireFeature;
import org.eblocker.server.common.malware.MalwareFilterService;
import org.eblocker.server.common.util.UrlUtils;
import org.eblocker.server.http.service.DeviceService;
import org.eblocker.server.icap.transaction.Transaction;
import org.eblocker.server.icap.transaction.TransactionProcessor;
import org.eblocker.registration.ProductFeature;
import org.eblocker.server.common.network.BaseURLs;
import com.google.common.base.Joiner;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.google.inject.name.Named;

@Singleton
@RequireFeature(ProductFeature.PRO)
public class MalwareFilterProcessor implements TransactionProcessor {

    private final String redirectPage;
    private final BaseURLs baseUrls;
    private final DeviceService deviceService;
    private final MalwareFilterService filterService;

    @Inject
    public MalwareFilterProcessor(@Named("malware.filter.redirect.page") String redirectPage,
                                  BaseURLs baseUrls,
                                  DeviceService deviceService,
                                  MalwareFilterService filterService) {
        this.redirectPage = redirectPage;
        this.baseUrls = baseUrls;
        this.deviceService = deviceService;
        this.filterService = filterService;
    }

    @Override
    public boolean process(Transaction transaction) {
        if (!deviceService.getDeviceById(transaction.getSession().getDeviceId()).isMalwareFilterEnabled()) {
            return true;
        }

        String url = transaction.getRequest().getUri();
        if (filterService.isBlocked(url)) {
            transaction.redirect(baseUrls.selectURLForPage(url)
                + redirectPage
                + "?target=" + UrlUtils.urlEncode(url)
                + "&malware=" + Joiner.on(",").join(filterService.getMalwareByUrl(url)));
            return false;
        }
        return true;
    }
}
